# coding:utf-8
"""
编写人：caoyong，时间：2022-11-23
该模块需要安装  pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple pymssql

"""
import pymssql
from caoyongpy import dbsql


class MsSql(dbsql.SqlData):  # sqlServer数据库类
    def __init__(self, server, user, password, database):
        """
        初始化数据库连接
        :param server: 数据库连接地址:端口 ,例如：10.10.3.55:1433
        :param user:  登录数据库的用户名
        :param password:  登录数据的密码
        :param database:   数据库名
        """
        self.db_type = 'pymssql'
        self.db = pymssql.connect(server, user, password, database)  # 打开数据库连接
        dbsql.SqlData.__init__(self, self.db, {"'": "''"})

    def close(self):  # 关闭数据库
        self.db.close()

    def commit(self):  # 提交事务
        self.db.commit()

    def rollback(self):  # 回滚事务
        self.db.rollback()
